package com.ricemall.common.valid;

import javax.validation.ConstraintValidator;
import javax.validation.ConstraintValidatorContext;
import java.util.HashSet;
import java.util.Set;

/**
 * @Author cappuccino
 * @Date 2021-06-07 12:06
 */
public class ListValueConstraintValidator implements ConstraintValidator<ListValue,Integer> {//<注解，校验值类型>

    /**
     * 存储所有可能的值
     */
    private Set<Integer> set = new HashSet<Integer>();

    /**
     * 初始化，你可以获取注解上的内容并进行处理
     * @param constraintAnnotation
     */
    @Override
    public void initialize(ListValue constraintAnnotation) {
        // 获取后端写好的限制
        // 这个value就是ListValue里的value，我们写的注解是@ListValue(value={0,1})
        int[] values = constraintAnnotation.vals();
//        if (values != null && values.length > 0) {
            for (int value : values) {
                set.add(value);
            }
//        }
    }

    /**
     * 覆写验证逻辑
     * @param value
     * @param context
     * @return
     */
    @Override
    public boolean isValid(Integer value, ConstraintValidatorContext context) {
        // 看是否在限制的值里
        return set.contains(value);
    }

}
